home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeViewKDE.py < prev    next >
Text File  |  2009-11-02  |  36KB  |  845 lines
  1. # DistUpgradeViewKDE.py 
  2. #  
  3. #  Copyright (c) 2007 Canonical Ltd
  4. #  
  5. #  Author: Jonathan Riddell <jriddell@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from PyQt4.QtCore import *
  23. from PyQt4.QtGui import *
  24. from PyQt4 import uic
  25.  
  26. import sys
  27. import logging
  28. import time
  29. import subprocess
  30. import traceback
  31. import tempfile
  32.  
  33. import apt
  34. import apt_pkg
  35. import os
  36. import shutil
  37.  
  38. import pty
  39.  
  40. from DistUpgradeApport import *
  41.  
  42. from DistUpgradeController import DistUpgradeController
  43. from DistUpgradeView import DistUpgradeView, FuzzyTimeToStr, InstallProgress, FetchProgress
  44.  
  45. import select
  46. import gettext
  47. from DistUpgradeGettext import gettext as gett
  48.  
  49. def _(str):
  50.     return unicode(gett(str), 'UTF-8')
  51.  
  52. def utf8(str):
  53.   if isinstance(str, unicode):
  54.       return str
  55.   return unicode(str, 'UTF-8')
  56.  
  57. def loadUi(file, parent):
  58.     if os.path.exists(file):
  59.         uic.loadUi(file, parent)
  60.     else:
  61.         #FIXME find file
  62.         print "error, can't find file: " + file
  63.  
  64. class DumbTerminal(QTextEdit):
  65.     """ A very dumb terminal """
  66.     def __init__(self, installProgress, parent_frame):
  67.         " really dumb terminal with simple editing support "
  68.         QTextEdit.__init__(self, "", parent_frame)
  69.         self.installProgress = installProgress
  70.         self.setFontFamily("Monospace")
  71.         self.setFontPointSize(8)
  72.         self.setWordWrapMode(QTextOption.NoWrap)
  73.         self.setUndoRedoEnabled(False)
  74.         self.setOverwriteMode(True)
  75.         self._block = False
  76.         #self.connect(self, SIGNAL("cursorPositionChanged()"), 
  77.         #             self.onCursorPositionChanged)
  78.  
  79.     def fork(self):
  80.         """pty voodoo"""
  81.         (self.child_pid, self.installProgress.master_fd) = pty.fork()
  82.         if self.child_pid == 0:
  83.             os.environ["TERM"] = "dumb"
  84.         return self.child_pid
  85.  
  86.     def updateInterface(self):
  87.         (rlist, wlist, xlist) = select.select([self.installProgress.master_fd],[],[], 0)
  88.         if len(rlist) > 0:
  89.             line = os.read(self.installProgress.master_fd, 255)
  90.             self.insertWithTermCodes(utf8(line))
  91.         QApplication.processEvents()
  92.  
  93.     def insertWithTermCodes(self, text):
  94.         """ support basic terminal codes """
  95.         display_text = ""
  96.         for c in text:
  97.             # \b - backspace - this seems to comes as "^H" now ??!
  98.             if ord(c) == 8:
  99.                 self.insertPlainText(display_text)
  100.                 self.textCursor().deletePreviousChar()
  101.                 display_text=""
  102.             # \r - is filtered out
  103.             elif c == chr(13):
  104.                 pass
  105.             # \a - bell - ignore for now
  106.             elif c == chr(7):
  107.                 pass
  108.             else:
  109.                 display_text += c
  110.         self.insertPlainText(display_text)
  111.  
  112.     def keyPressEvent(self, ev):
  113.         """ send (ascii) key events to the pty """
  114.         # no master_fd yet
  115.         if not hasattr(self.installProgress, "master_fd"):
  116.             return
  117.         # special handling for backspace
  118.         if ev.key() == Qt.Key_Backspace:
  119.             #print "sent backspace"
  120.             os.write(self.installProgress.master_fd, chr(8))
  121.             return
  122.         # do nothing for events like "shift" 
  123.         if not ev.text():
  124.             return
  125.         # now sent the key event to the termianl as utf-8
  126.         os.write(self.installProgress.master_fd, ev.text().toUtf8())
  127.  
  128.     def onCursorPositionChanged(self):
  129.         """ helper that ensures that the cursor is always at the end """
  130.         if self._block:
  131.             return
  132.         # block signals so that we do not run into a recursion
  133.         self._block = True
  134.         self.moveCursor(QTextCursor.End)
  135.         self._block = False
  136.  
  137. class KDECdromProgressAdapter(apt.progress.CdromProgress):
  138.     """ Report the cdrom add progress """
  139.     def __init__(self, parent):
  140.         self.status = parent.window_main.label_status
  141.         self.progressbar = parent.window_main.progressbar_cache
  142.         self.parent = parent
  143.  
  144.     def update(self, text, step):
  145.         """ update is called regularly so that the gui can be redrawn """
  146.         if text:
  147.           self.status.setText(text)
  148.         self.progressbar.setValue(step/float(self.totalSteps))
  149.         QApplication.processEvents()
  150.  
  151.     def askCdromName(self):
  152.         return (False, "")
  153.  
  154.     def changeCdrom(self):
  155.         return False
  156.  
  157. class KDEOpProgress(apt.progress.OpProgress):
  158.   """ methods on the progress bar """
  159.   def __init__(self, progressbar, progressbar_label):
  160.       self.progressbar = progressbar
  161.       self.progressbar_label = progressbar_label
  162.       #self.progressbar.set_pulse_step(0.01)
  163.       #self.progressbar.pulse()
  164.  
  165.   def update(self, percent):
  166.       #if percent > 99:
  167.       #    self.progressbar.set_fraction(1)
  168.       #else:
  169.       #    self.progressbar.pulse()
  170.       #self.progressbar.set_fraction(percent/100.0)
  171.       self.progressbar.setValue(percent)
  172.       QApplication.processEvents()
  173.  
  174.   def done(self):
  175.       self.progressbar_label.setText("")
  176.  
  177. class KDEFetchProgressAdapter(FetchProgress):
  178.     """ methods for updating the progress bar while fetching packages """
  179.     # FIXME: we really should have some sort of "we are at step"
  180.     # xy in the gui
  181.     # FIXME2: we need to thing about mediaCheck here too
  182.     def __init__(self, parent):
  183.         FetchProgress.__init__(self)
  184.         # if this is set to false the download will cancel
  185.         self.status = parent.window_main.label_status
  186.         self.progress = parent.window_main.progressbar_cache
  187.         self.parent = parent
  188.  
  189.     def mediaChange(self, medium, drive):
  190.       msg = _("Please insert '%s' into the drive '%s'") % (medium,drive)
  191.       change = QMessageBox.question(self.parent.window_main, _("Media Change"), msg, QMessageBox.Ok, QMessageBox.Cancel)
  192.       if change == QMessageBox.Ok:
  193.         return True
  194.       return False
  195.  
  196.     def start(self):
  197.         FetchProgress.start(self)
  198.         #self.progress.show()
  199.         self.progress.setValue(0)
  200.         self.status.show()
  201.  
  202.     def stop(self):
  203.         self.parent.window_main.progress_text.setText("  ")
  204.         self.status.setText(_("Fetching is complete"))
  205.  
  206.     def pulse(self):
  207.         """ we don't have a mainloop in this application, we just call processEvents here and elsewhere"""
  208.         # FIXME: move the status_str and progress_str into python-apt
  209.         # (python-apt need i18n first for this)
  210.         FetchProgress.pulse(self)
  211.         self.progress.setValue(self.percent)
  212.         currentItem = self.currentItems + 1
  213.         if currentItem > self.totalItems:
  214.             currentItem = self.totalItems
  215.  
  216.         if self.currentCPS > 0:
  217.             self.status.setText(_("Fetching file %li of %li at %sB/s") % (currentItem, self.totalItems, apt_pkg.SizeToStr(self.currentCPS)))
  218.             self.parent.window_main.progress_text.setText("<i>" + _("About %s remaining") % unicode(FuzzyTimeToStr(self.eta), 'utf-8') + "</i>")
  219.         else:
  220.             self.status.setText(_("Fetching file %li of %li") % (currentItem, self.totalItems))
  221.             self.parent.window_main.progress_text.setText("  ")
  222.  
  223.         QApplication.processEvents()
  224.         return True
  225.  
  226. class KDEInstallProgressAdapter(InstallProgress):
  227.     """methods for updating the progress bar while installing packages"""
  228.     # timeout with no status change when the terminal is expanded
  229.     # automatically
  230.     TIMEOUT_TERMINAL_ACTIVITY = 240
  231.  
  232.     def __init__(self,parent):
  233.         InstallProgress.__init__(self)
  234.         self._cache = None
  235.         self.label_status = parent.window_main.label_status
  236.         self.progress = parent.window_main.progressbar_cache
  237.         self.progress_text = parent.window_main.progress_text
  238.         self.parent = parent
  239.         try:
  240.             self._terminal_log = open("/var/log/dist-upgrade/term.log","w")
  241.         except Exception, e:
  242.             # if something goes wrong (permission denied etc), use stdout
  243.             logging.error("Can not open terminal log: '%s'" % e)
  244.             self._terminal_log = sys.stdout
  245.         # some options for dpkg to make it die less easily
  246.         apt_pkg.Config.Set("DPkg::StopOnError","False")
  247.  
  248.     def startUpdate(self):
  249.         InstallProgress.startUpdate(self)
  250.         self.finished = False
  251.         # FIXME: add support for the timeout
  252.         # of the terminal (to display something useful then)
  253.         # -> longer term, move this code into python-apt 
  254.         self.label_status.setText(_("Applying changes"))
  255.         self.progress.setValue(0)
  256.         self.progress_text.setText(" ")
  257.         # do a bit of time-keeping
  258.         self.start_time = 0.0
  259.         self.time_ui = 0.0
  260.         self.last_activity = 0.0
  261.         self.parent.window_main.showTerminalButton.setEnabled(True)
  262.  
  263.     def error(self, pkg, errormsg):
  264.         InstallProgress.error(self, pkg, errormsg)
  265.         logging.error("got an error from dpkg for pkg: '%s': '%s'" % (pkg, errormsg))
  266.         # we do not report followup errors from earlier failures
  267.         if gettext.dgettext('dpkg', "dependency problems - leaving unconfigured") in errormsg:
  268.           return False
  269.         summary = _("Could not install '%s'") % pkg
  270.         msg = _("The upgrade will continue but the '%s' package may not "
  271.                 "be in a working state. Please consider submitting a "
  272.                 "bug report about it.") % pkg
  273.         msg = "<big><b>%s</b></big><br />%s" % (summary, msg)
  274.  
  275.         dialogue = QDialog(self.parent.window_main)
  276.         loadUi("dialog_error.ui", dialogue)
  277.         self.parent.translate_widget_children(dialogue)
  278.         dialogue.label_error.setText(utf8(msg))
  279.         if errormsg != None:
  280.             dialogue.textview_error.setText(utf8(errormsg))
  281.             dialogue.textview_error.show()
  282.         else:
  283.             dialogue.textview_error.hide()
  284.         dialogue.connect(dialogue.button_bugreport, SIGNAL("clicked()"), self.parent.reportBug)
  285.         dialogue.exec_()
  286.  
  287.     def conffile(self, current, new):
  288.         """ask question in case conffile has been changed by user"""
  289.         logging.debug("got a conffile-prompt from dpkg for file: '%s'" % current)
  290.         start = time.time()
  291.         prim = _("Replace the customized configuration file\n'%s'?") % current
  292.         sec = _("You will lose any changes you have made to this "
  293.                 "configuration file if you choose to replace it with "
  294.                 "a newer version.")
  295.         markup = "<span weight=\"bold\" size=\"larger\">%s </span> \n\n%s" % (prim, sec)
  296.         self.confDialogue = QDialog(self.parent.window_main)
  297.         loadUi("dialog_conffile.ui", self.confDialogue)
  298.         self.confDialogue.label_conffile.setText(markup)
  299.         self.confDialogue.textview_conffile.hide()
  300.         #FIXME, below to be tested
  301.         #self.confDialogue.resize(self.confDialogue.minimumSizeHint())
  302.         self.confDialogue.connect(self.confDialogue.show_difference_button, SIGNAL("clicked()"), self.showConffile)
  303.  
  304.         # now get the diff
  305.         if os.path.exists("/usr/bin/diff"):
  306.           cmd = ["/usr/bin/diff", "-u", current, new]
  307.           diff = utf8(subprocess.Popen(cmd, stdout=subprocess.PIPE).communicate()[0])
  308.           self.confDialogue.textview_conffile.setText(diff)
  309.         else:
  310.           self.confDialogue.textview_conffile.setText(_("The 'diff' command was not found"))
  311.         result = self.confDialogue.exec_()
  312.         self.time_ui += time.time() - start
  313.         # if replace, send this to the terminal
  314.         if result == QDialog.Accepted:
  315.             os.write(self.master_fd, "y\n")
  316.         else:
  317.             os.write(self.master_fd, "n\n")
  318.  
  319.     def showConffile(self):
  320.         if self.confDialogue.textview_conffile.isVisible():
  321.             self.confDialogue.textview_conffile.hide()
  322.             self.confDialogue.show_difference_button.setText(_("Show Difference >>>"))
  323.         else:
  324.             self.confDialogue.textview_conffile.show()
  325.             self.confDialogue.show_difference_button.setText(_("<<< Hide Difference"))
  326.  
  327.     def fork(self):
  328.         """pty voodoo"""
  329.         (self.child_pid, self.master_fd) = pty.fork()
  330.         if self.child_pid == 0:
  331.             os.environ["TERM"] = "dumb"
  332.             if (not os.environ.has_key("DEBIAN_FRONTEND") or
  333.                 os.environ["DEBIAN_FRONTEND"] == "kde"):
  334.                 os.environ["DEBIAN_FRONTEND"] = "noninteractive"
  335.             os.environ["APT_LISTCHANGES_FRONTEND"] = "none"
  336.         logging.debug(" fork pid is: %s" % self.child_pid)
  337.         return self.child_pid
  338.  
  339.     def statusChange(self, pkg, percent, status):
  340.         """update progress bar and label"""
  341.         # start the timer when the first package changes its status
  342.         if self.start_time == 0.0:
  343.           #print "setting start time to %s" % self.start_time
  344.           self.start_time = time.time()
  345.         self.progress.setValue(self.percent)
  346.         self.label_status.setText(unicode(status.strip(), 'UTF-8'))
  347.         # start showing when we gathered some data
  348.         if percent > 1.0:
  349.           self.last_activity = time.time()
  350.           self.activity_timeout_reported = False
  351.           delta = self.last_activity - self.start_time
  352.           # time wasted in conffile questions (or other ui activity)
  353.           delta -= self.time_ui
  354.           time_per_percent = (float(delta)/percent)
  355.           eta = (100.0 - self.percent) * time_per_percent
  356.           # only show if we have some sensible data (60sec < eta < 2days)
  357.           if eta > 61.0 and eta < (60*60*24*2):
  358.             self.progress_text.setText(_("About %s remaining") % FuzzyTimeToStr(eta))
  359.           else:
  360.             self.progress_text.setText(" ")
  361.  
  362.     def finishUpdate(self):
  363.         self.label_status.setText("")
  364.  
  365.     def updateInterface(self):
  366.         """
  367.         no mainloop in this application, just call processEvents lots here
  368.         it's also important to sleep for a minimum amount of time
  369.         """
  370.         # log the output of dpkg (on the master_fd) to the terminal log
  371.         while True:
  372.             try:
  373.                 (rlist, wlist, xlist) = select.select([self.master_fd],[],[], 0)
  374.                 if len(rlist) > 0:
  375.                     line = os.read(self.master_fd, 255)
  376.                     self._terminal_log.write(line)
  377.                     self.parent.terminal_text.insertWithTermCodes(utf8(line))
  378.                 else:
  379.                     break
  380.             except Exception, e:
  381.                 print e
  382.                 logging.debug("error reading from self.master_fd '%s'" % e)
  383.                 break
  384.  
  385.         # now update the GUI
  386.         try:
  387.           InstallProgress.updateInterface(self)
  388.         except ValueError, e:
  389.           logging.error("got ValueError from InstallProgress.updateInterface. Line was '%s' (%s)" % (self.read, e))
  390.           # reset self.read so that it can continue reading and does not loop
  391.           self.read = ""
  392.         # check about terminal activity
  393.         if self.last_activity > 0 and \
  394.            (self.last_activity + self.TIMEOUT_TERMINAL_ACTIVITY) < time.time():
  395.           if not self.activity_timeout_reported:
  396.             #FIXME bug 95465, I can't recreate this, so here's a hacky fix
  397.             try:
  398.                 logging.warning("no activity on terminal for %s seconds (%s)" % (self.TIMEOUT_TERMINAL_ACTIVITY, self.label_status.text()))
  399.             except UnicodeEncodeError:
  400.                 logging.warning("no activity on terminal for %s seconds" % (self.TIMEOUT_TERMINAL_ACTIVITY))
  401.             self.activity_timeout_reported = True
  402.           self.parent.window_main.konsole_frame.show()
  403.         QApplication.processEvents()
  404.         time.sleep(0.02)
  405.  
  406.     def waitChild(self):
  407.         while True:
  408.             self.updateInterface()
  409.             (pid, res) = os.waitpid(self.child_pid,os.WNOHANG)
  410.             if pid == self.child_pid:
  411.                 break
  412.         # we need the full status here (not just WEXITSTATUS)
  413.         return res
  414.  
  415. # inherit from the class created in window_main.ui
  416. # to add the handler for closing the window
  417. class UpgraderMainWindow(QWidget):
  418.  
  419.     def __init__(self):
  420.         QWidget.__init__(self)
  421.         #uic.loadUi("window_main.ui", self)
  422.         loadUi("window_main.ui", self)
  423.  
  424.     def setParent(self, parentRef):
  425.         self.parent = parentRef
  426.  
  427.     def closeEvent(self, event):
  428.         close = self.parent.on_window_main_delete_event()
  429.         if close:
  430.             event.accept()
  431.         else:
  432.             event.ignore()
  433.  
  434. class DistUpgradeViewKDE(DistUpgradeView):
  435.     """KDE frontend of the distUpgrade tool"""
  436.     def __init__(self, datadir=None, logdir=None):
  437.         # silence the PyQt4 logger
  438.         logger = logging.getLogger("PyQt4")
  439.         logger.setLevel(logging.INFO)
  440.         if not datadir:
  441.           localedir=os.path.join(os.getcwd(),"mo")
  442.         else:
  443.           localedir="/usr/share/locale/update-manager"
  444.  
  445.         # FIXME: i18n must be somewhere relative do this dir
  446.         try:
  447.           gettext.bindtextdomain("update-manager", localedir)
  448.           gettext.textdomain("update-manager")
  449.         except Exception, e:
  450.           logging.warning("Error setting locales (%s)" % e)
  451.  
  452.         #about = KAboutData("adept_manager","Upgrader","0.1","Dist Upgrade Tool for Kubuntu",KAboutData.License_GPL,"(c) 2007 Canonical Ltd",
  453.         #"http://wiki.kubuntu.org/KubuntuUpdateManager", "jriddell@ubuntu.com")
  454.         #about.addAuthor("Jonathan Riddell", None,"jriddell@ubuntu.com")
  455.         #about.addAuthor("Michael Vogt", None,"michael.vogt@ubuntu.com")
  456.         #KCmdLineArgs.init(["./dist-upgrade.py"],about)
  457.  
  458.         # we test for DISPLAY here, QApplication does not throw a 
  459.         # exception when run without DISPLAY but dies instead
  460.         if not "DISPLAY" in os.environ:
  461.             raise Exception, "No DISPLAY in os.environ found"
  462.         self.app = QApplication(["update-manager"])
  463.  
  464.         if os.path.exists("/usr/share/icons/oxygen/48x48/apps/system-software-update.png"):
  465.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/apps/system-software-update.png")
  466.         else:
  467.             messageIcon = QPixmap("/usr/share/icons/hicolor/48x48/apps/adept_manager.png")
  468.         self.app.setWindowIcon(QIcon(messageIcon))
  469.  
  470.         self.window_main = UpgraderMainWindow()
  471.         self.window_main.setParent(self)
  472.         self.window_main.show()
  473.  
  474.         self.prev_step = 0 # keep a record of the latest step
  475.  
  476.         self._opCacheProgress = KDEOpProgress(self.window_main.progressbar_cache, self.window_main.progress_text)
  477.         self._fetchProgress = KDEFetchProgressAdapter(self)
  478.         self._cdromProgress = KDECdromProgressAdapter(self)
  479.  
  480.         self._installProgress = KDEInstallProgressAdapter(self)
  481.  
  482.         # reasonable fault handler
  483.         sys.excepthook = self._handleException
  484.  
  485.         self.window_main.showTerminalButton.setEnabled(False)
  486.         self.app.connect(self.window_main.showTerminalButton, SIGNAL("clicked()"), self.showTerminal)
  487.  
  488.         #kdesu requires us to copy the xauthority file before it removes it when Adept is killed
  489.         copyXauth = tempfile.mktemp("", "adept")
  490.         if 'XAUTHORITY' in os.environ and os.environ['XAUTHORITY'] != copyXauth:
  491.             shutil.copy(os.environ['XAUTHORITY'], copyXauth)
  492.             os.environ["XAUTHORITY"] = copyXauth
  493.  
  494.         # Note that with kdesudo this needs --nonewdcop
  495.         ## create a new DCOP-Client:
  496.         #client = DCOPClient()
  497.         ## connect the client to the local DCOP-server:
  498.         #client.attach()
  499.  
  500.         #for qcstring_app in client.registeredApplications():
  501.         #    app = str(qcstring_app)
  502.         #    if app.startswith("adept"): 
  503.         #        adept = DCOPApp(qcstring_app, client)
  504.         #        adeptInterface = adept.object("MainApplication-Interface")
  505.         #        adeptInterface.quit()
  506.  
  507.         # This works just as well
  508.         subprocess.call(["killall", "adept_manager"])
  509.         subprocess.call(["killall", "adept_updater"])
  510.  
  511.         # init gettext
  512.         gettext.bindtextdomain("update-manager",localedir)
  513.         gettext.textdomain("update-manager")
  514.         self.translate_widget_children()
  515.         self.window_main.label_title.setText(self.window_main.label_title.text().replace("Ubuntu", "Kubuntu"))
  516.  
  517.         # setup terminal text in hidden by default spot
  518.         self.window_main.konsole_frame.hide()
  519.         self.konsole_frame_layout = QHBoxLayout(self.window_main.konsole_frame)
  520.         self.window_main.konsole_frame.setMinimumSize(600, 400)
  521.         self.terminal_text = DumbTerminal(self._installProgress, self.window_main.konsole_frame)
  522.         self.konsole_frame_layout.addWidget(self.terminal_text)
  523.         self.terminal_text.show()
  524.  
  525.         # for some reason we need to start the main loop to get everything displayed
  526.         # this app mostly works with processEvents but run main loop briefly to keep it happily displaying all widgets
  527.         QTimer.singleShot(10, self.exitMainLoop)
  528.         self.app.exec_()
  529.  
  530.     def exitMainLoop(self):
  531.         print "exitMainLoop"
  532.         self.app.exit()
  533.  
  534.     def translate_widget_children(self, parentWidget=None):
  535.         if parentWidget == None:
  536.             parentWidget = self.window_main
  537.         if isinstance(parentWidget, QDialog) or isinstance(parentWidget, QWidget):
  538.             if str(parentWidget.windowTitle()) == "Error":
  539.                 parentWidget.setWindowTitle( gettext.dgettext("kdelibs", "Error"))
  540.             else:
  541.                 parentWidget.setWindowTitle(_( str(parentWidget.windowTitle()) ))
  542.  
  543.         if parentWidget.children() != None:
  544.             for widget in parentWidget.children():
  545.                 self.translate_widget(widget)
  546.                 self.translate_widget_children(widget)
  547.  
  548.     def translate_widget(self, widget):
  549.         if isinstance(widget, QLabel) or isinstance(widget, QPushButton):
  550.             if str(widget.text()) == "&Cancel":
  551.                 widget.setText(unicode(gettext.dgettext("kdelibs", "&Cancel"), 'UTF-8'))
  552.             elif str(widget.text()) == "&Close":
  553.                 widget.setText(unicode(gettext.dgettext("kdelibs", "&Close"), 'UTF-8'))
  554.             elif str(widget.text()) != "":
  555.                 widget.setText( _(str(widget.text())).replace("_", "&") )
  556.  
  557.     def _handleException(self, exctype, excvalue, exctb):
  558.         """Crash handler."""
  559.  
  560.         if (issubclass(exctype, KeyboardInterrupt) or
  561.             issubclass(exctype, SystemExit)):
  562.             return
  563.  
  564.         # we handle the exception here, hand it to apport and run the
  565.         # apport gui manually after it because we kill u-m during the upgrade
  566.         # to prevent it from popping up for reboot notifications or FF restart
  567.         # notifications or somesuch
  568.         lines = traceback.format_exception(exctype, excvalue, exctb)
  569.         logging.error("not handled exception in KDE frontend:\n%s" % "\n".join(lines))
  570.         # we can't be sure that apport will run in the middle of a upgrade
  571.         # so we still show a error message here
  572.         apport_crash(exctype, excvalue, exctb)
  573.         if not run_apport():
  574.             tbtext = ''.join(traceback.format_exception(exctype, excvalue, exctb))
  575.             dialog = QDialog(self.window_main)
  576.             loadUi("dialog_error.ui", dialog)
  577.             self.translate_widget_children(self.dialog)
  578.             #FIXME make URL work
  579.             #dialog.connect(dialog.beastie_url, SIGNAL("leftClickedURL(const QString&)"), self.openURL)
  580.             dialog.crash_detail.setText(tbtext)
  581.             dialog.exec_()
  582.         sys.exit(1)
  583.  
  584.     def openURL(self, url):
  585.         """start konqueror"""
  586.         #need to run this else kdesu can't run Konqueror
  587.         #subprocess.call(['su', 'ubuntu', 'xhost', '+localhost'])
  588.         QDesktopServices.openUrl(QUrl(url))
  589.  
  590.     def reportBug(self):
  591.         """start konqueror"""
  592.         #need to run this else kdesu can't run Konqueror
  593.         #subprocess.call(['su', 'ubuntu', 'xhost', '+localhost'])
  594.         QDesktopServices.openUrl(QUrl("https://launchpad.net/ubuntu/+source/update-manager/+filebug"))
  595.  
  596.     def showTerminal(self):
  597.         if self.window_main.konsole_frame.isVisible():
  598.             self.window_main.konsole_frame.hide()
  599.             self.window_main.showTerminalButton.setText(_("Show Terminal >>>"))
  600.         else:
  601.             self.window_main.konsole_frame.show()
  602.             self.window_main.showTerminalButton.setText(_("<<< Hide Terminal"))
  603.         self.window_main.resize(self.window_main.sizeHint())
  604.  
  605.     def getFetchProgress(self):
  606.         return self._fetchProgress
  607.  
  608.     def getInstallProgress(self, cache):
  609.         self._installProgress._cache = cache
  610.         return self._installProgress
  611.  
  612.     def getOpCacheProgress(self):
  613.         return self._opCacheProgress
  614.  
  615.     def getCdromProgress(self):
  616.         return self._cdromProgress
  617.  
  618.     def updateStatus(self, msg):
  619.         self.window_main.label_status.setText(utf8(msg))
  620.  
  621.     def hideStep(self, step):
  622.         image = getattr(self.window_main,"image_step%i" % step)
  623.         label = getattr(self.window_main,"label_step%i" % step)
  624.         image.hide()
  625.         label.hide()
  626.  
  627.     def abort(self):
  628.         step = self.prev_step
  629.         if step > 0:
  630.             image = getattr(self.window_main,"image_step%i" % step)
  631.             if os.path.exists("/usr/share/icons/oxygen/16x16/actions/dialog-cancel.png"):
  632.                 cancelIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/dialog-cancel.png")
  633.             elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-cancel.png"):
  634.                 cancelIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-cancel.png")
  635.             else:
  636.                 cancelIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/cancel.png")
  637.             image.setPixmap(cancelIcon)
  638.             image.show()
  639.  
  640.     def setStep(self, step):
  641.         if os.path.exists("/usr/share/icons/oxygen/16x16/actions/dialog-ok.png"):
  642.             okIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/dialog-ok.png")
  643.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-ok.png"):
  644.             okIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/dialog-ok.png")
  645.         else:
  646.             okIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/ok.png")
  647.  
  648.         if os.path.exists("/usr/share/icons/oxygen/16x16/actions/arrow-right.png"):
  649.             arrowIcon = QPixmap("/usr/share/icons/oxygen/16x16/actions/arrow-right.png")
  650.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/16x16/actions/arrow-right.png"):
  651.             arrowIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/16x16/actions/arrow-right.png")
  652.         else:
  653.             arrowIcon = QPixmap("/usr/share/icons/crystalsvg/16x16/actions/1rightarrow.png")
  654.  
  655.         if self.prev_step:
  656.             image = getattr(self.window_main,"image_step%i" % self.prev_step)
  657.             label = getattr(self.window_main,"label_step%i" % self.prev_step)
  658.             image.setPixmap(okIcon)
  659.             image.show()
  660.             ##arrow.hide()
  661.         self.prev_step = step
  662.         # show the an arrow for the current step and make the label bold
  663.         image = getattr(self.window_main,"image_step%i" % step)
  664.         label = getattr(self.window_main,"label_step%i" % step)
  665.         image.setPixmap(arrowIcon)
  666.         image.show()
  667.         label.setText("<b>" + label.text() + "</b>")
  668.  
  669.     def information(self, summary, msg, extended_msg=None):
  670.         msg = "<big><b>%s</b></big><br />%s" % (summary,msg)
  671.  
  672.         dialogue = QDialog(self.window_main)
  673.         loadUi("dialog_error.ui", dialogue)
  674.         self.translate_widget_children(dialogue)
  675.         dialogue.label_error.setText(utf8(msg))
  676.         if extended_msg != None:
  677.             dialogue.textview_error.setText(utf8(extended_msg))
  678.             dialogue.textview_error.show()
  679.         else:
  680.             dialogue.textview_error.hide()
  681.         dialogue.button_bugreport.hide()
  682.         dialogue.setWindowTitle(_("Information"))
  683.  
  684.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-information.png"):
  685.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-information.png")
  686.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-information.png"):
  687.             messageIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-information.png")
  688.         else:
  689.             messageIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_info.png")
  690.         dialogue.image.setPixmap(messageIcon)
  691.         dialogue.exec_()
  692.  
  693.     def error(self, summary, msg, extended_msg=None):
  694.         msg="<big><b>%s</b></big><br />%s" % (summary, msg)
  695.  
  696.         dialogue = QDialog(self.window_main)
  697.         loadUi("dialog_error.ui", dialogue)
  698.         self.translate_widget_children(dialogue)
  699.         dialogue.label_error.setText(utf8(msg))
  700.         if extended_msg != None:
  701.             dialogue.textview_error.setText(utf8(extended_msg))
  702.             dialogue.textview_error.show()
  703.         else:
  704.             dialogue.textview_error.hide()
  705.         dialogue.button_close.show()
  706.         self.app.connect(dialogue.button_bugreport, SIGNAL("clicked()"), self.reportBug)
  707.  
  708.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-error.png"):
  709.             messageIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-error.png")
  710.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-error.png"):
  711.             messageIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-error.png")
  712.         else:
  713.             messageIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_critical.png")
  714.         dialogue.image.setPixmap(messageIcon)
  715.         dialogue.exec_()
  716.  
  717.         return False
  718.  
  719.     def confirmChanges(self, summary, changes, downloadSize, 
  720.                        actions=None, removal_bold=True):
  721.         """show the changes dialogue"""
  722.         # FIXME: add a whitelist here for packages that we expect to be
  723.         # removed (how to calc this automatically?)
  724.         DistUpgradeView.confirmChanges(self, summary, changes, downloadSize)
  725.         msg = unicode(self.confirmChangesMessage, 'UTF-8')
  726.         self.changesDialogue = QDialog(self.window_main)
  727.         loadUi("dialog_changes.ui", self.changesDialogue)
  728.  
  729.         self.changesDialogue.treeview_details.hide()
  730.         self.changesDialogue.connect(self.changesDialogue.show_details_button, SIGNAL("clicked()"), self.showChangesDialogueDetails)
  731.         self.translate_widget_children(self.changesDialogue)
  732.         self.changesDialogue.show_details_button.setText(_("Details") + " >>>")
  733.         self.changesDialogue.resize(self.changesDialogue.sizeHint())
  734.  
  735.         if os.path.exists("/usr/share/icons/oxygen/48x48/status/dialog-warning.png"):
  736.             warningIcon = QPixmap("/usr/share/icons/oxygen/48x48/status/dialog-warning.png")
  737.         elif os.path.exists("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-warning.png"):
  738.             warningIcon = QPixmap("/usr/lib/kde4/share/icons/oxygen/48x48/status/dialog-warning.png")
  739.         else:
  740.             warningIcon = QPixmap("/usr/share/icons/crystalsvg/32x32/actions/messagebox_warning.png")
  741.  
  742.         self.changesDialogue.question_pixmap.setPixmap(warningIcon)
  743.  
  744.         if actions != None:
  745.             cancel = actions[0].replace("_", "")
  746.             self.changesDialogue.button_cancel_changes.setText(cancel)
  747.             confirm = actions[1].replace("_", "")
  748.             self.changesDialogue.button_confirm_changes.setText(confirm)
  749.  
  750.         summaryText = unicode("<big><b>%s</b></big>" % summary, 'UTF-8')
  751.         self.changesDialogue.label_summary.setText(summaryText)
  752.         self.changesDialogue.label_changes.setText(msg)
  753.         # fill in the details
  754.         self.changesDialogue.treeview_details.clear()
  755.         self.changesDialogue.treeview_details.setHeaderLabels(["Packages"])
  756.         self.changesDialogue.treeview_details.header().hide()
  757.         for rm in self.toRemove:
  758.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Remove %s") % rm]) )
  759.         for inst in self.toInstall:
  760.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Install %s") % inst]) )
  761.         for up in self.toUpgrade:
  762.             self.changesDialogue.treeview_details.insertTopLevelItem(0, QTreeWidgetItem(self.changesDialogue.treeview_details, [_("Upgrade %s") % up]) )
  763.  
  764.         #FIXME resize label, stop it being shrinkable
  765.         res = self.changesDialogue.exec_()
  766.         if res == QDialog.Accepted:
  767.             return True
  768.         return False
  769.  
  770.     def showChangesDialogueDetails(self):
  771.         if self.changesDialogue.treeview_details.isVisible():
  772.             self.changesDialogue.treeview_details.hide()
  773.             self.changesDialogue.show_details_button.setText(_("Details") + " >>>")
  774.         else:
  775.             self.changesDialogue.treeview_details.show()
  776.             self.changesDialogue.show_details_button.setText("<<< " + _("Details"))
  777.         self.changesDialogue.resize(self.changesDialogue.sizeHint())
  778.  
  779.     def askYesNoQuestion(self, summary, msg, default='No'):
  780.         answer = QMessageBox.question(self.window_main, unicode(summary, 'UTF-8'), unicode("<font>") + unicode(msg, 'UTF-8'), QMessageBox.Yes|QMessageBox.No, QMessageBox.No)
  781.         if answer == QMessageBox.Yes:
  782.             return True
  783.         return False
  784.  
  785.     def confirmRestart(self):
  786.         messageBox = QMessageBox(QMessageBox.Question, _("Restart required"), _("<b><big>Restart the system to complete the upgrade</big></b>"), QMessageBox.NoButton, self.window_main)
  787.         yesButton = messageBox.addButton(QMessageBox.Yes)
  788.         noButton = messageBox.addButton(QMessageBox.No)
  789.         yesButton.setText(_("_Restart Now").replace("_", "&"))
  790.         noButton.setText(gettext.dgettext("kdelibs", "&Close"))
  791.         answer = messageBox.exec_()
  792.         if answer == QMessageBox.Yes:
  793.             return True
  794.         return False
  795.  
  796.     def processEvents(self):
  797.         QApplication.processEvents()
  798.  
  799.     def on_window_main_delete_event(self):
  800.         #FIXME make this user friendly
  801.         text = _("""<b><big>Cancel the running upgrade?</big></b>
  802.  
  803. The system could be in an unusable state if you cancel the upgrade. You are strongly advised to resume the upgrade.""")
  804.         text = text.replace("\n", "<br />")
  805.         cancel = QMessageBox.warning(self.window_main, _("Cancel Upgrade?"), text, QMessageBox.Yes, QMessageBox.No)
  806.         if cancel == QMessageBox.Yes:
  807.             return True
  808.         return False
  809.  
  810. if __name__ == "__main__":
  811.   
  812.   view = DistUpgradeViewKDE()
  813.   view.askYesNoQuestion("input box test","bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar bar ")
  814.  
  815.   if sys.argv[1] == "--test-term":
  816.       pid = view.terminal_text.fork()
  817.       if pid == 0:
  818.           subprocess.call(["bash"])
  819.           sys.exit()
  820.       while True:
  821.           view.terminal_text.updateInterface()
  822.           QApplication.processEvents()
  823.           time.sleep(0.01)
  824.  
  825.   if sys.argv[1] == "--show-in-terminal":
  826.       for c in open(sys.argv[2]).read():
  827.           view.terminal_text.insertWithTermCodes( c )
  828.           #print c, ord(c)
  829.           QApplication.processEvents()
  830.           time.sleep(0.05)
  831.       while True:
  832.           QApplication.processEvents()
  833.  
  834.   cache = apt.Cache()
  835.   for pkg in sys.argv[1:]:
  836.     if cache[pkg].isInstalled and not cache[pkg].isUpgradable: 
  837.       cache[pkg].markDelete(purge=True)
  838.     else:
  839.       cache[pkg].markInstall()
  840.   cache.commit(view._fetchProgress,view._installProgress)
  841.  
  842.   # keep the window open
  843.   while True:
  844.       QApplication.processEvents()
  845.